package com.itic.codeGen;

import java.io.File;
import java.io.IOException;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.mybatis.generator.api.MyBatisGenerator;
import org.mybatis.generator.config.Configuration;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.exception.InvalidConfigurationException;
import org.mybatis.generator.exception.XMLParserException;
import org.mybatis.generator.internal.DefaultShellCallback;

import com.itic.codeGen.codeGenerator.util.ParameterUtil;
import com.itic.codeGen.codeGenerator.util.CommonUtils.FILE_TYPE;
import com.itic.codeGen.codeGenerator.workLine.FileGenerator;
import com.itic.codeGen.codeGenerator.workLine.TargetJava;
import com.itic.codeGen.codeGenerator.workLine.TargetXML;

public class BaseGenerator {
    public static void main(String[] args) {
	// my batis
	List<String> warnings = new ArrayList<String>();
	ConfigurationParser cp = new ConfigurationParser(warnings);
	boolean overwrite = true;
	// staticTableConfig.xml,dynamicTableConfig.xml
	File configFile = new File("src/main/resources/generatorConfig.xml");
	try {
	    System.out.println("Loading Conf File....");
	    Configuration config = cp.parseConfiguration(configFile);
	    DefaultShellCallback callback = new DefaultShellCallback(overwrite);
	    System.out.println("Connect DB......");
	    MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config,
		    callback, warnings);
	    System.out.println("Starting Generating...");
	    myBatisGenerator.generate(null);
	    System.out.println("Warnings...");
	    for(String warn:warnings){
		System.out.println(warn);
	    }
	    // Parameters
	    Map<String, Object> params = ParameterUtil.getJavaParams(config
		    .getContexts().get(0));
	    TargetJava javaFile = new TargetJava(params);
	    TargetXML xmlFile = new TargetXML(params);
	    FileGenerator generator = new FileGenerator();
	    // generator Action file
	    //javaFile.setFileType(params, FILE_TYPE.CONTROLLER);
	   // generator.generateClass(javaFile);
	    // service
	    //javaFile.setFileType(params, FILE_TYPE.SERVICE);
	    //generator.generateClass(javaFile);
	    // dao
	    javaFile.setFileType(params, FILE_TYPE.DAO);
	    generator.generateInterface(javaFile);
	    // impl
	    /*
	     * tj.setFileType(params, FILE_TYPE.IMPL); fg.generateClass(tj);
	     */
	    // xml
	    xmlFile.setFileType(params, FILE_TYPE.XML);
	    generator.generateXML(xmlFile);

	    System.out.println("Job DONE!! Refreash");
	} catch (IOException e) {
	    e.printStackTrace();
	} catch (XMLParserException e) {
	    e.printStackTrace();
	} catch (InvalidConfigurationException e) {
	    e.printStackTrace();
	} catch (SQLException e) {
	    e.printStackTrace();
	} catch (InterruptedException e) {
	    e.printStackTrace();
	}
    }
}
